package ru.ifmo.mlarinsky.classificator.classification;

import java.awt.Color;
import java.util.HashMap;
import java.util.Map;
import ru.ifmo.mlarinsky.classificator.io.InstrumentLoader;

/**
 * @author Mikhail Larinskiy
 */
public class DJClassification extends Classification {
    private static final Map<String, String> INDEX_TO_SECTOR = new HashMap<String, String>();
    private static final Map<String, Color> INDEX_TO_COLOR = new HashMap<String, Color>();
    static {
        INDEX_TO_SECTOR.put("DJUSBM", "Business Materials");
        INDEX_TO_SECTOR.put("DJUSNC", "Consumer Goods");
        INDEX_TO_SECTOR.put("DJUSCY", "Consumer Services");
        INDEX_TO_SECTOR.put("DJUSFN", "Financials");
        INDEX_TO_SECTOR.put("DJUSHC", "Health Care");
        INDEX_TO_SECTOR.put("DJUSIN", "Industrial");
        INDEX_TO_SECTOR.put("DJUSEN", "Oil & Gas");
        INDEX_TO_SECTOR.put("DJUSTC", "Technology");
        INDEX_TO_SECTOR.put("DJUSTL", "Telecommunications");
        INDEX_TO_SECTOR.put("DJUSUT", "Utilities");

        INDEX_TO_COLOR.put("DJUSBM", new Color(255, 0, 0));
        INDEX_TO_COLOR.put("DJUSNC", new Color(255, 178, 0));
        INDEX_TO_COLOR.put("DJUSCY", new Color(255, 255, 0));
        INDEX_TO_COLOR.put("DJUSFN", new Color(0, 255, 0));
        INDEX_TO_COLOR.put("DJUSHC", new Color(0, 255, 255));
        INDEX_TO_COLOR.put("DJUSIN", new Color(0, 0, 255));
        INDEX_TO_COLOR.put("DJUSEN", new Color(178, 0, 255));
        INDEX_TO_COLOR.put("DJUSTC", new Color(255, 0, 215));
        INDEX_TO_COLOR.put("DJUSTL", new Color(178, 255, 0));
        INDEX_TO_COLOR.put("DJUSUT", new Color(255, 100, 100));
    }

    public DJClassification() {
        super(InstrumentLoader.DJ_INDEXES, INDEX_TO_SECTOR, INDEX_TO_COLOR);
    }
}
